pytorch tensor to numpy dtype